package main

import (
	"database/sql"
	"errors"
	"fmt"
	"gorm.io/gen"
	"gorm.io/gorm"
	"jgyApi/server/global"
	"strings"
	"time"
)

type Model struct {
	UUID uint      `gorm:"primaryKey"`
	Time time.Time `gorm:"column:test_time"`
}
type TestUser struct {
	Model        Model      `gorm:"embedded"`
	Name         string     `gorm:"index;not null"`
	Email        *string    `gorm:"not null"`
	Age          uint8      `gorm:"not null;default:0;"`
	Birthday     *time.Time `gorm:"not null"`
	MemberNumber sql.NullString
	ActivatedAt  sql.NullTime
}

func TestCreateTable() {
	err := global.GVA_DB.AutoMigrate(&TestUser{})
	if err != nil {
		fmt.Println(err)
	}
}

func TestCreate() {
	var email string
	var birthday time.Time
	birthday = time.Now()
	email = "2607771759@qq.com"
	//创建一条 结构体方式
	//result := global.GVA_DB.Create(&TestUser{Name: "bob", Email: &email, Birthday: &birthday, Model: Model{Time: birthday}})
	//创建一条 map方式
	//result := global.GVA_DB.Model(&TestUser{}).Create(map[string]interface{}{
	//	"Name": "bob2","Email":&email,"Birthday":&birthday,
	//
	//})
	//创建多条 map方式
	//result := global.GVA_DB.Model(&TestUser{}).Create([]map[string]interface{}{
	//	{"Name":"bob3", "Email":&email, "Birthday":&birthday},
	//	{"Name":"bob4", "Email":&email, "Birthday":&birthday},
	//	{"Name":"bob5", "Email":&email, "Birthday":&birthday},
	//})
	//创建多条 结构体方式
	result := global.GVA_DB.Create(&[]TestUser{
		{Name: "bob6", Email: &email, Birthday: &birthday, Model: Model{Time: birthday}},
		{Name: "bob7", Email: &email, Birthday: &birthday, Model: Model{Time: birthday}},
		{Name: "bob8", Email: &email, Birthday: &birthday}, //这条没有Model的话其中的 Model中的Time字段会为0000-00-00 00:00:00.000
	})
	fmt.Println(result.Error)
	fmt.Println(result.RowsAffected)

}

func TestQuery() {
	//var user TestUser
	var users []TestUser

	//select * from test_user order by uuid limit 1
	//result := global.GVA_DB.First(&user)

	//result := global.GVA_DB.First(&user, 3)  //select * from test_user where uuid=3 order by uuid limit 1
	result := global.GVA_DB.Find(&users, []int{1, 2, 3})

	fmt.Println(result.RowsAffected)
	fmt.Println(users)
	if result.Error != nil {
		// 检查 ErrRecordNotFound 错误
		errors.Is(result.Error, gorm.ErrRecordNotFound)
	}
}

func TestUpdate() {
	//1.使用save进行更新
	//var user TestUser
	//result := global.GVA_DB.First(&user)
	//user.Name = "homie"
	//global.GVA_DB.Save(&user)
	////有primaryKey的情况下更新指定这条
	//result = global.GVA_DB.Select("Name").Save(&TestUser{Model: Model{UUID: 2}, Name: "mia"})

	//2.使用where进行更新
	result := global.GVA_DB.Model(&TestUser{}).Where("uuid = ?", 5).Update("name", "lynn")
	if result.Error != nil {
		// 检查 ErrRecordNotFound 错误
		errors.Is(result.Error, gorm.ErrRecordNotFound)
	}

}

func TestDel()  {
	result := global.GVA_DB.Where("uuid = ?", 9).Delete(&TestUser{})
	if result.Error != nil {
		// 检查 ErrRecordNotFound 错误
		errors.Is(result.Error, gorm.ErrRecordNotFound)
	}
}


func TestGEN() {

	// 生成实例
	g := gen.NewGenerator(gen.Config{
		// 相对执行`go run`时的路径, 会自动创建目录
		OutPath: "./query",

		// WithDefaultQuery 生成默认查询结构体(作为全局变量使用), 即`Q`结构体和其字段(各表模型)
		// WithoutContext 生成没有context调用限制的代码供查询
		// WithQueryInterface 生成interface形式的查询代码(可导出), 如`Where()`方法返回的就是一个可导出的接口类型
		Mode: gen.WithDefaultQuery | gen.WithQueryInterface,

		// 表字段可为 null 值时, 对应结体字段使用指针类型
		FieldNullable: true, // generate pointer when field is nullable

		// 表字段默认值与模型结构体字段零值不一致的字段, 在插入数据时需要赋值该字段值为零值的, 结构体字段须是指针类型才能成功, 即`FieldCoverable:true`配置下生成的结构体字段.
		// 因为在插入时遇到字段为零值的会被GORM赋予默认值. 如字段`age`表默认值为10, 即使你显式设置为0最后也会被GORM设为10提交.
		// 如果该字段没有上面提到的插入时赋零值的特殊需要, 则字段为非指针类型使用起来会比较方便.
		FieldCoverable: false, // generate pointer when field has default value, to fix problem zero value cannot be assign: https://gorm.io/docs/create.html#Default-Values

		// 模型结构体字段的数字类型的符号表示是否与表字段的一致, `false`指示都用有符号类型
		FieldSignable: false, // detect integer field's unsigned type, adjust generated data type
		// 生成 gorm 标签的字段索引属性
		FieldWithIndexTag: false, // generate with gorm index tag
		// 生成 gorm 标签的字段类型属性
		FieldWithTypeTag: true, // generate with gorm column type tag
	})
	// 设置目标 db
	g.UseDB(global.GVA_DB)

	// 自定义字段的数据类型
	// 统一数字类型为int64,兼容protobuf
	//type map[string]func(columnType gorm.ColumnType) (dataType string)
	dataMap := map[string]func(detailType gorm.ColumnType) (dataType string){
		"tinyint":   func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"smallint":  func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"mediumint": func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"bigint":    func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"int":       func(detailType gorm.ColumnType) (dataType string) { return "int64" },
	}
	// 要先于`ApplyBasic`执行
	g.WithDataTypeMap(dataMap)

	// 自定义模型结体字段的标签
	// 将特定字段名的 json 标签加上`string`属性,即 MarshalJSON 时该字段由数字类型转成字符串类型
	jsonField := gen.FieldJSONTagWithNS(func(columnName string) (tagContent string) {
		toStringField := `balance, `
		if strings.Contains(toStringField, columnName) {
			return columnName + ",string"
		}
		return columnName
	})
	// 将非默认字段名的字段定义为自动时间戳和软删除字段;
	// 自动时间戳默认字段名为:`updated_at`、`created_at, 表字段数据类型为: INT 或 DATETIME
	// 软删除默认字段名为:`deleted_at`, 表字段数据类型为: DATETIME
	//type func(tag field.GormTag) field.GormTag
	//autoUpdateTimeField := gen.FieldGORMTag("update_time", "column:update_time;type:int unsigned;autoUpdateTime")
	//autoCreateTimeField := gen.FieldGORMTag("create_time", "column:create_time;type:int unsigned;autoCreateTime")
	//softDeleteField := gen.FieldType("delete_time", "soft_delete.del")
	// 模型自定义选项组
	//fieldOpts := []gen.ModelOpt{jsonField, autoCreateTimeField, autoUpdateTimeField, softDeleteField}
	fieldOpts := []gen.ModelOpt{jsonField}

	// 创建模型的结构体,生成文件在 model 目录; 先创建的结果会被后面创建的覆盖
	// 这里创建个别模型仅仅是为了拿到`*generate.QueryStructMeta`类型对象用于后面的模型关联操作中
	//User := g.GenerateModel("user")
	// 创建全部模型文件, 并覆盖前面创建的同名模型
	allModel := g.GenerateAllTable(fieldOpts...)

	// 创建有关联关系的模型文件
	// 可以用于指定外键
	//Score := g.GenerateModel("score",
	//	append(
	//		fieldOpts,
	//		// user 一对多 address 关联, 外键`uid`在 address 表中
	//		gen.FieldRelate(field.HasMany, "user", User, &field.RelateConfig{GORMTag: "foreignKey:UID"}),
	//	)...,
	//)

	// 创建模型的方法,生成文件在 query 目录; 先创建结果不会被后创建的覆盖
	//g.ApplyBasic(User)
	g.ApplyBasic(allModel...)

	g.Execute()
}


func GenSpecialModel(name string) {
	// 生成实例
	g := gen.NewGenerator(gen.Config{
		// 相对执行`go run`时的路径, 会自动创建目录
		OutPath: "./query",

		// WithDefaultQuery 生成默认查询结构体(作为全局变量使用), 即`Q`结构体和其字段(各表模型)
		// WithoutContext 生成没有context调用限制的代码供查询
		// WithQueryInterface 生成interface形式的查询代码(可导出), 如`Where()`方法返回的就是一个可导出的接口类型
		Mode: gen.WithDefaultQuery | gen.WithQueryInterface,

		// 表字段可为 null 值时, 对应结体字段使用指针类型
		FieldNullable: true, // generate pointer when field is nullable

		// 表字段默认值与模型结构体字段零值不一致的字段, 在插入数据时需要赋值该字段值为零值的, 结构体字段须是指针类型才能成功, 即`FieldCoverable:true`配置下生成的结构体字段.
		// 因为在插入时遇到字段为零值的会被GORM赋予默认值. 如字段`age`表默认值为10, 即使你显式设置为0最后也会被GORM设为10提交.
		// 如果该字段没有上面提到的插入时赋零值的特殊需要, 则字段为非指针类型使用起来会比较方便.
		FieldCoverable: false, // generate pointer when field has default value, to fix problem zero value cannot be assign: https://gorm.io/docs/create.html#Default-Values

		// 模型结构体字段的数字类型的符号表示是否与表字段的一致, `false`指示都用有符号类型
		FieldSignable: false, // detect integer field's unsigned type, adjust generated data type
		// 生成 gorm 标签的字段索引属性
		FieldWithIndexTag: false, // generate with gorm index tag
		// 生成 gorm 标签的字段类型属性
		FieldWithTypeTag: true, // generate with gorm column type tag
	})
	// 设置目标 db
	g.UseDB(global.GVA_DB)
	// 自定义字段的数据类型
	// 统一数字类型为int64,兼容protobuf
	//type map[string]func(columnType gorm.ColumnType) (dataType string)
	dataMap := map[string]func(detailType gorm.ColumnType) (dataType string){
		"tinyint":   func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"smallint":  func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"mediumint": func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"bigint":    func(detailType gorm.ColumnType) (dataType string) { return "int64" },
		"int":       func(detailType gorm.ColumnType) (dataType string) { return "int64" },
	}
	// 要先于`ApplyBasic`执行
	g.WithDataTypeMap(dataMap)
	// 自定义模型结体字段的标签
	// 将特定字段名的 json 标签加上`string`属性,即 MarshalJSON 时该字段由数字类型转成字符串类型
	jsonField := gen.FieldJSONTagWithNS(func(columnName string) (tagContent string) {
		toStringField := `balance, `
		if strings.Contains(toStringField, columnName) {
			return columnName + ",string"
		}
		return columnName
	})
	// 将非默认字段名的字段定义为自动时间戳和软删除字段;
	// 自动时间戳默认字段名为:`updated_at`、`created_at, 表字段数据类型为: INT 或 DATETIME
	// 软删除默认字段名为:`deleted_at`, 表字段数据类型为: DATETIME
	//type func(tag field.GormTag) field.GormTag
	//autoUpdateTimeField := gen.FieldGORMTag("update_time", "column:update_time;type:int unsigned;autoUpdateTime")
	//autoCreateTimeField := gen.FieldGORMTag("create_time", "column:create_time;type:int unsigned;autoCreateTime")
	//softDeleteField := gen.FieldType("delete_time", "soft_delete.del")
	// 模型自定义选项组
	//fieldOpts := []gen.ModelOpt{jsonField, autoCreateTimeField, autoUpdateTimeField, softDeleteField}
	//fieldOpts := []gen.ModelOpt{jsonField}

	// 创建模型的结构体,生成文件在 model 目录; 先创建的结果会被后面创建的覆盖
	// 这里创建个别模型仅仅是为了拿到`*generate.QueryStructMeta`类型对象用于后面的模型关联操作中
	Model := g.GenerateModel(name, jsonField)
	// 创建全部模型文件, 并覆盖前面创建的同名模型

	// 创建有关联关系的模型文件
	// 可以用于指定外键
	//Score := g.GenerateModel("score",
	//	append(
	//		fieldOpts,
	//		// user 一对多 address 关联, 外键`uid`在 address 表中
	//		gen.FieldRelate(field.HasMany, "user", User, &field.RelateConfig{GORMTag: "foreignKey:UID"}),
	//	)...,
	//)
	// 创建模型的方法,生成文件在 query 目录; 先创建结果不会被后创建的覆盖
	g.ApplyBasic(Model)
	g.Execute()
}

func implodeAndExplode()  {
	strArrayOld:=[]string{"aa","bb"}
	//implode
	strTotal:=strings.Join(strArrayOld,"-")

	//explode
	strArrayNew:=strings.Split(strTotal,"-")
	fmt.Println(strArrayNew)

}

